POJ 3259 Wormholes（Bellman－Ford判断是否有负权边）

//
//  main.cpp
//  Richard
//
//  Created by 邵金杰 on 16/8/12.
//  Copyright © 2016年 邵金杰. All rights reserved.
//


<pre name="code" class="cpp">#include<cstdio>
#include<vector>
using namespace std;
#define INF 1<<30
struct edges{
    int s,e,w;
    edges(int s,int e,int w): s(s),e(e),w(w) {}
};
const int maxn=500+50;
int dist[maxn];
vector<edges> v;
bool Bellman_Ford(int n)
{
    for(int i=1;i<=n;i++)
        dist[i]=INF;
    dist[1]=0;
    for(int k=1;k<n;k++)
    {
        for(int i=0;i<v.size();i++)
        {
            int s=v[i].s;
            int e=v[i].e;
            int w=v[i].w;
            if(dist[s]+w<dist[e]) dist[e]=dist[s]+w;
        }
    }
    for(int i=0;i<v.size();i++)
    {
        int s=v[i].s;
        int e=v[i].e;
        int w=v[i].w;
        if(dist[s]+w<dist[e]) return true;
    }
    return false;
}
int main()
{
    int F;
    scanf("%d",&F);
    while(F--)
    {
        v.clear();
        int n,m,w;
        int s,e,t;
        scanf("%d%d%d",&n,&m,&w);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&s,&e,&t);
            v.push_back(edges(s,e,t));
            v.push_back(edges(e,s,t));
        }
        for(int i=0;i<w;i++)
        {
            scanf("%d%d%d",&s,&e,&t);
            v.push_back(edges(s,e,-t));
        }
        if(Bellman_Ford(n)) printf("YES\n");
        else printf("NO\n");
    }
}